package com.theory.oss.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.CannedAccessControlList;
import com.theory.oss.service.FileService;
import com.theory.oss.util.OssProperties;
import org.joda.time.DateTime;
import org.springframework.stereotype.Service;

import java.io.InputStream;
import java.util.UUID;

@Service
public class FileServiceImpl implements FileService {

    /**
     *
     * @param inputStream 文件输入流
     * @param module 模块名称，相当于文件夹，如头像，图标等
     * @param fileName 原始文件名
     * @return 返回上传后的url地址
     */
    @Override
    public String upload(InputStream inputStream, String module, String fileName) {
        // 创建OSSClient实例
        OSS ossClient = new OSSClientBuilder().build(
                OssProperties.ENDPOINT,
                OssProperties.KEY_ID,
                OssProperties.KEY_SECRET);

        // 判断oss实例是否存在：如果不存在则创建，如果存在则获取
        if (!ossClient.doesBucketExist(OssProperties.BUCKET_NAME)) {
            // 创建bucket
            ossClient.createBucket(OssProperties.BUCKET_NAME);
            // 设置oss实例的访问权限为公共读
            ossClient.setBucketAcl(OssProperties.BUCKET_NAME, CannedAccessControlList.PublicRead);
        }

        // 构建日期路径: avatar/2021/10/3/文件名
        String folder = new DateTime().toString("yyyy/MM/dd");

        // 文件名: uuid.扩展名
        fileName = UUID.randomUUID().toString() + fileName.substring(fileName.lastIndexOf("."));

        // 文件根路径
        String key = module + "/" + folder + "/" + fileName;

        // 文件上传至阿里云
        ossClient.putObject(OssProperties.BUCKET_NAME, key, inputStream);

        // 关闭OSSClient
        ossClient.shutdown();

        // 返回阿里云文件绝对路径
//        https://codetheory-srb.oss-cn-shenzhen.aliyuncs.com/snipaste_20211003_115551.png
        return "https://" + OssProperties.BUCKET_NAME + "." + OssProperties.ENDPOINT + "/" + key;

    }

    /**
     * 根据路径删除文件
     * @param url 这里前台传过来的是一个完整的文件url地址，例：https://codetheory-srb.oss-cn-shenzhen.aliyuncs.com/snipaste_20211003_115551.png
     *            但是我们只需要最后面的文件名即可，所以要把前缀去掉
     */
    @Override
    public void removeFile(String url) {
        // 创建OSSClient实例
        OSS ossClient = new OSSClientBuilder().build(
                OssProperties.ENDPOINT,
                OssProperties.KEY_ID,
                OssProperties.KEY_SECRET);

        //文件名路径（服务器上的文件路径）
        String host = "https://" + OssProperties.BUCKET_NAME + "." + OssProperties.ENDPOINT + "/";

        // 获取文件名
        String objectName = url.substring(host.length());

        // 删除文件
        ossClient.deleteObject(OssProperties.BUCKET_NAME, objectName);

        // 关闭OSSClient
        ossClient.shutdown();

    }
}
